HERMES: A Multi-Granularity Labeling Substrate for Pre-training Data Mixtures
HERMES:面向预训练数据混合的多粒度分层标注基板
🧭 预训练数据混合 · 多粒度标签
Wizard Quant · Peking University · University of Science and Technology of China
提出一种基于残差矢量量化的多粒度数据标注框架,使预训练数据混合从固定标签转变为可重用粒度层次导航。
📌 工作参考价值(4 点)
- 在电商推荐模型训练中,可借鉴 HERMES 的语义嵌入 + 残差矢量量化构建商品或用户行为的多粒度类别体系,替代固定的 taxonomy,实现 training data mixture 的动态粒度调节。
- 实验中发现的“粗粒度下等比例覆盖 + 质量前 30% 混合规则有效,细粒度下因候选池收缩而失效”现象,提示在推荐数据采样时需根据当前粒度调整采样策略,避免直接在极细粒度下套用同一种比例规则。
- 该层次标签基板可复用于推荐模型的预训练、微调或持续训练阶段,作为 data loader 的输入,按不同前缀长度控制样本分布,探索多任务能力 Lift。
- 工程实现上,3 阶段 RVQ 编码开销可控,可离线一次性为全量数据集标注,后续只需通过前缀截断切换粒度,适合工业级海量数据场景。